package cn.hy.controller;

import org.redisson.api.RBitSet;
import org.redisson.api.RedissonClient;
import org.springframework.dao.DataAccessException;
import org.springframework.data.redis.connection.BitFieldSubCommands;
import org.springframework.data.redis.connection.RedisConnection;
import org.springframework.data.redis.core.RedisCallback;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import java.text.SimpleDateFormat;
import java.time.Instant;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.ZoneId;
import java.time.format.DateTimeFormatter;
import java.util.Date;
import java.util.List;

@RestController
@RequestMapping("/sign")
public class SignRedissonController {

    @Resource
    private RedissonClient redissonClient;

    /**
     * 当日签到
     *
     * @return
     */
    @PostMapping("/do")
    public String doSign() {

        //用户Uid 从用户中心获取
        String userId = "1234";
        //获取当前日期
        LocalDateTime now = LocalDateTime.now();
        String keySuffix = now.format(DateTimeFormatter.ofPattern(":yyyyMM"));
        // 组合成key= sign:userId:年月
        String key = "sign:" + userId + keySuffix;
        // 获取当前日期是当月的第多少天
        int dayOfMonth = now.getDayOfMonth();
        RBitSet bitSet = redissonClient.getBitSet(key);
        // 存到redis中的bitmap中，由于bitmap从0开始，第多少天从1开始，dayOfMonth需要减1
        bitSet.set(dayOfMonth - 1, true);
        return "ok";
    }


    /**
     * 指定日期补签
     *
     * @return
     */
    @PostMapping("/doSign")
    public String doSupplementarySignature(@RequestParam String signDate) {
        String userId = "1234";
        LocalDateTime dateOfSign = stringParseLocalDateTime(signDate);
        String keySuffix = dateOfSign.format(DateTimeFormatter.ofPattern(":yyyyMM"));
        String key = "sign:" + userId + keySuffix;
        int dayOfMonth = dateOfSign.getDayOfMonth();
        RBitSet bitSet = redissonClient.getBitSet(key);
        bitSet.set(dayOfMonth - 1, true);
        return "ok";
    }

    /**
     * 统计指定月的用户总共签到次数
     *  date：年月日
     * @return
     */
    @GetMapping("/count")
    public Integer getSignCount(@RequestParam String date) {
        String userId = "1234";
        LocalDateTime dateOfSign = stringParseLocalDateTime(date);
        String keySuffix = dateOfSign.format(DateTimeFormatter.ofPattern(":yyyyMM"));
        String key = "sign:" + userId + keySuffix;
        RBitSet bitSet = redissonClient.getBitSet(key);
        return (int)bitSet.size();
    }

    /**
     * 统计当前日期往前的连续签到数（连续的）
     *
     * @return
     */
    @GetMapping("/signCount")
    public Integer signCount() {
        //用户Uid
        String userId = "1234";
        //获取当前日期
        LocalDateTime now = LocalDateTime.now();
        String keySuffix = now.format(DateTimeFormatter.ofPattern(":yyyyMM"));
        // 组合成key= sign:userId:年月
        String key = "sign:" + userId + keySuffix;
        // 获取当前日期是当月的第多少天
        int dayOfMonth = now.getDayOfMonth();
        // 获取本月截止今天为止的所有签到记录，返回的是一个十进制的数字
        // 一个月不能超出 31天， int bit 最大32 位 妙
        RBitSet bitSet = redissonClient.getBitSet(key);
        long result = bitSet.getUnsigned(dayOfMonth, 0);
        if (result == 0) return 0;
        int count = 0;
        while (true) {
            // 让这个数字与1做与运算，得到数字的最后一位bit，并判断这个bit是否为0
            if ((result & 1) == 0) {
                // 如果为0说明不是连续签到，直接终止
                break;
            } else {
                // 如果不为0，那就为1，说明有签到，继续下次循环，并且计数器加一
                count++;
            }
            // 把数字右移一位，抛弃最后一位的bit，继续下一个bit
            result = result >>> 1; // 也可以写成 number >>>= 1
        }
        return count;
    }

    /**
     * 字符串日期转换成LocalDateTime
     */
    private LocalDateTime stringParseLocalDateTime(String date) {
        try {
            SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyyMMdd");
            Date parse = simpleDateFormat.parse(date);
            Instant instant = parse.toInstant();
            ZoneId zoneId = ZoneId.systemDefault();
            return instant.atZone(zoneId).toLocalDateTime();
        } catch (Exception e) {
            e.printStackTrace();
            throw new RuntimeException("日期格式转换报错");
        }
    }

    public static void main(String[] args) {
        int dayOfMonth = LocalDate.now().getDayOfMonth();

        System.out.println(BitFieldSubCommands.BitFieldType.unsigned(dayOfMonth));
    }

}
